#pragma once

namespace keras2cpp
{
	class KerasEnsembleModel
	{
	public:
		KerasEnsembleModel(const std::string& directPath, const int& threadCount);
		~KerasEnsembleModel();

	public:
		std::vector<float> predict(const std::vector<float>& inputParamVec);

	private:
		std::vector<float> getAverage(std::vector<std::vector<float>>& PredictValues);

	private:
		std::vector<std::vector<std::shared_ptr<KerasModel>>> vModel;
		int modelCount;
	};

}